package org.eztouch.zuul.filter;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.lang3.StringUtils;

import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;

/**
 * Zuul过滤器<br>
 * 
 * 实现了在请求被路由之前检查请求中是否有accessToken参数，若有就进行路由，若没有就拒绝访问，返回401 Unauthorized错误。<br>
 * <br>
 * 
 * 实现了自定义过滤器之后，还需要实例化该过滤器才能生效:应用主类中添加该过滤器的bean<br>
 * 	@Bean
	public TokenFilter getTokenFilter(){
		return new TokenFilter();
	}
 * 
 * @author eztouch
 *
 */
public class TokenFilter extends ZuulFilter {
	
	/**
	 * 返回一个字符串代表过滤器的类型，在zuul中定义了四种不同生命周期的过滤器类型，具体如下：
			pre：可以在请求被路由之前调用
			routing：在路由请求时候被调用
			post：在routing和error过滤器之后被调用
			error：处理请求时发生错误时被调用
	 */
	@Override
	public String filterType() {
		return "pre";
	}
	
	/**
	 * 通过int值来定义过滤器的执行顺序
	 */
	@Override
	public int filterOrder() {
		return 0;
	}
	
	/**
	 * 返回一个boolean类型来判断该过滤器是否要执行，所以通过此函数可实现过滤器的开关。
	 */
	@Override
	public boolean shouldFilter() {
		return true;
	}
	
	/**
	 * 过滤器的具体逻辑。
	 * 
	 * 需要注意，这里我们通过ctx.setSendZuulResponse(false)令zuul过滤该请求，不对其进行路由，
	 * 然后通过ctx.setResponseStatusCode(401)设置了其返回的错误码，
	 * 
	 * 当然我们也可以进一步优化我们的返回，比如，通过ctx.setResponseBody(body)对返回body内容进行编辑等。
	 */
	@Override
	public Object run() {
		RequestContext ctx = RequestContext.getCurrentContext();
		HttpServletRequest request = ctx.getRequest();
		
		System.out.println("请求方式："+request.getMethod()+"，请求地址："+request.getRequestURI());
		
		String tokenKey = request.getHeader("authorization");
		if (StringUtils.isNotBlank(tokenKey)) {
			System.out.println("请求Token不为空："+tokenKey);
		}else{
			tokenKey = request.getParameter("authorization");
			if(StringUtils.isBlank(tokenKey)){
				ctx.setSendZuulResponse(false);
				ctx.setResponseStatusCode(401);
				return null;
			}
		}
		return null;
	}

	



	

}
